# -*- coding: utf-8 -*-

from svmutil import *

# y,x = svm_read_problem("../test/train")
# yt,xt = svm_read_problem("../test/test")
# model = svm_train(y, x)
# save_model = svm_save_model("../test/model", model)
# print('test:')
# p_label, p_acc, p_val = svm_predict(yt[200:201], xt[200:201], model)
# print(p_label)

y, x = svm_read_problem("train_file")
model = svm_train(y, x)
svm_save_model("model", model)
print()

